package stack;

import java.util.Stack;

/**
* 给定一个整数数组 asteroids，表示在同一行的小行星。

对于数组中的每一个元素，其绝对值表示小行星的大小，正负表示小行星的移动方向（正表示向右移动，负表示向左移动）。每一颗小行星以相同的速度移动。

找出碰撞后剩下的所有小行星。碰撞规则：两个小行星相互碰撞，较小的小行星会爆炸。如果两颗小行星大小相同，则两颗小行星都会爆炸。两颗移动方向相同的小行星，永远不会发生碰撞。asteroids[i] != 0
 */
public class AsteroidCollision {
    public int[] asteroidCollision(int[] asteroids) {
        //将asteroids中得数字按照正负区分
        Stack<Integer> stack = new Stack<>();

        for(int asteroid : asteroids){
            int flag = 0;
            //正数入栈
            if(asteroid > 0){
                stack.push(asteroid);
            }else{
                //判断当前是否会有碰撞的机会
                while(stack.size() != 0 && flag == 0){
                    //根据大小判断谁爆炸
                    if(Math.abs(stack.peek()) > Math.abs(asteroid)){
                        flag = 1;
                    }else if(Math.abs(stack.peek()) < Math.abs(asteroid)){
                        stack.pop();
                    }else if(Math.abs(stack.peek()) == Math.abs(asteroid)){
                        stack.pop();
                        flag = 1;
                    }
                }
            }
        }
        int[] ints = new int[stack.size()];
        for (int i = ints.length -1; i >=0 ; i--) {
            ints[i] = stack.pop();
        }
        return ints;
    }
}
